package com.leyantech.ecp.agenx.task.filter.takeaction;

import com.leyantech.ai.DialogueReply;
import com.leyantech.ecp.agenx.task.filter.RateLimiter;

import com.google.inject.Singleton;

/**
 * @author eahau, {@literal <eahau@foxmail.com>}
 * @date 2022-11-15.
 */
@Singleton
public class TakeActionRateLimiter extends RateLimiter<DialogueReply> {

  @Override
  protected String[] getRaterKeys(DialogueReply dialogueReply) {
    final String sid = dialogueReply.getStoreId();
    final String aid = dialogueReply.getAssistantId();
    return new String[]{sid, aid};
  }

}
